home *** CD-ROM | disk | FTP | other *** search
/ PC World Komputer 2010 April / PCWorld0410.iso / hity wydania / Ubuntu 9.10 PL / karmelkowy-koliberek-desktop-9.10-i386-PL.iso / casper / filesystem.squashfs / usr / share / hplip / base / slp.pyc (.txt) < prev    next >
Encoding:
Python Compiled Bytecode  |  2009-10-28  |  4.2 KB  |  167 lines
  1. # Source Generated with Decompyle++
  2. # File: in.pyc (Python 2.6)
  3.  
  4. import sys
  5. import time
  6. import socket
  7. import select
  8. import struct
  9. import random
  10. import re
  11. from g import *
  12. import utils
  13. prod_pat = re.compile('\\(\\s*x-hp-prod_id\\s*=\\s*(.*?)\\s*\\)', re.IGNORECASE)
  14. mac_pat = re.compile('\\(\\s*x-hp-mac\\s*=\\s*(.*?)\\s*\\)', re.IGNORECASE)
  15. num_port_pat = re.compile('\\(\\s*x-hp-num_port\\s*=\\s*(.*?)\\s*\\)', re.IGNORECASE)
  16. ip_pat = re.compile('\\(\\s*x-hp-ip\\s*=\\s*(.*?)\\s*\\)', re.IGNORECASE)
  17. p1_pat = re.compile('\\(\\s*x-hp-p1\\s*=(?:\\d\\)|\\s*(.*?)\\s*\\))', re.IGNORECASE)
  18. p2_pat = re.compile('\\(\\s*x-hp-p2\\s*=(?:\\d\\)|\\s*(.*?)\\s*\\))', re.IGNORECASE)
  19. p3_pat = re.compile('\\(\\s*x-hp-p3\\s*=(?:\\d\\)|\\s*(.*?)\\s*\\))', re.IGNORECASE)
  20. hn_pat = re.compile('\\(\\s*x-hp-hn\\s*=\\s*(.*?)\\s*\\)', re.IGNORECASE)
  21.  
  22. def detectNetworkDevices(ttl = 4, timeout = 10):
  23.     (mcast_addr, mcast_port) = ('224.0.1.60', 427)
  24.     found_devices = { }
  25.     s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, socket.IPPROTO_UDP)
  26.     x = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
  27.     
  28.     try:
  29.         x.connect(('1.2.3.4', 56))
  30.     except socket.error:
  31.         log.error('Network is unreachable. Please check your network connection and try again.')
  32.         return { }
  33.  
  34.     intf = x.getsockname()[0]
  35.     x.close()
  36.     s.setblocking(0)
  37.     ttl = struct.pack('B', ttl)
  38.     
  39.     try:
  40.         s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
  41.         s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEPORT, 1)
  42.     except (AttributeError, socket.error):
  43.         pass
  44.  
  45.     
  46.     try:
  47.         s.setsockopt(socket.SOL_IP, socket.IP_MULTICAST_TTL, ttl)
  48.         s.setsockopt(socket.SOL_IP, socket.IP_MULTICAST_IF, socket.inet_aton(intf) + socket.inet_aton('0.0.0.0'))
  49.         s.setsockopt(socket.SOL_IP, socket.IP_MULTICAST_LOOP, 1)
  50.     except Exception:
  51.         e = None
  52.         log.error('Unable to setup multicast socket for SLP: %s' % e)
  53.         return { }
  54.  
  55.     packet = ''.join([
  56.         '\x01\x06\x00,\x00\x00en\x00\x03',
  57.         struct.pack('!H', random.randint(1, 65535)),
  58.         '\x00\x00\x00\x18service:x-hpnp-discover:\x00\x00\x00\x00'])
  59.     
  60.     try:
  61.         s.sendto(packet, 0, (mcast_addr, mcast_port))
  62.     except socket.error:
  63.         e = None
  64.         log.error('Unable to send broadcast SLP packet: %s' % e)
  65.  
  66.     time_left = timeout
  67.     while time_left > 0:
  68.         start_time = time.time()
  69.         (r, w, e) = select.select([
  70.             s], [], [
  71.             s], time_left)
  72.         time_left -= time.time() - start_time
  73.         if not r:
  74.             continue
  75.         
  76.         (data, addr) = s.recvfrom(2048)
  77.         update_spinner()
  78.         log.log_data(data, width = 32)
  79.         
  80.         try:
  81.             (ver, func, length, flags, dialect, lang_code, char_encode, recv_xid, status_code, attr_length) = struct.unpack('!BBHBBHHHHH', data[:16])
  82.             x = struct.unpack('!%ds' % attr_length, data[16:])[0].strip()
  83.         except struct.error:
  84.             continue
  85.  
  86.         
  87.         try:
  88.             num_ports = int(num_port_pat.search(x).group(1))
  89.         except (AttributeError, ValueError):
  90.             num_ports = 1
  91.  
  92.         if num_ports == 0:
  93.             num_ports = 1
  94.         
  95.         y = {
  96.             'num_devices': 0,
  97.             'num_ports': num_ports,
  98.             'product_id': '',
  99.             'status_code': 0,
  100.             'device2': '0',
  101.             'device3': '0',
  102.             'note': '',
  103.             'device1': '0' }
  104.         
  105.         try:
  106.             y['device1'] = p1_pat.search(x).group(1)
  107.         except AttributeError:
  108.             y['device1'] = '0'
  109.  
  110.         y['num_devices'] += 1
  111.         if num_ports > 1:
  112.             
  113.             try:
  114.                 y['device2'] = p2_pat.search(x).group(1)
  115.             except AttributeError:
  116.                 y['device2'] = '0'
  117.  
  118.             y['num_devices'] += 1
  119.             if num_ports > 2:
  120.                 
  121.                 try:
  122.                     y['device3'] = p3_pat.search(x).group(1)
  123.                 except AttributeError:
  124.                     y['device3'] = '0'
  125.  
  126.                 y['num_devices'] += 1
  127.             
  128.         
  129.         if y['device1'] is None:
  130.             y['device1'] = '0'
  131.         
  132.         if y['device2'] is None:
  133.             y['device2'] = '0'
  134.         
  135.         if y['device3'] is None:
  136.             y['device3'] = '0'
  137.         
  138.         
  139.         try:
  140.             y['product_id'] = prod_pat.search(x).group(1)
  141.         except AttributeError:
  142.             y['product_id'] = ''
  143.  
  144.         
  145.         try:
  146.             y['mac'] = mac_pat.search(x).group(1)
  147.         except AttributeError:
  148.             y['mac'] = ''
  149.  
  150.         
  151.         try:
  152.             y['ip'] = ip_pat.search(x).group(1)
  153.         except AttributeError:
  154.             y['ip'] = ''
  155.  
  156.         
  157.         try:
  158.             y['hn'] = hn_pat.search(x).group(1)
  159.         except AttributeError:
  160.             y['hn'] = ''
  161.  
  162.         y['status_code'] = status_code
  163.         found_devices[addr[0]] = y
  164.         log.debug('Found device: %s' % y)
  165.     return found_devices
  166.  
  167.